# model
model_name = '20241115'
use_checkpoint = None
feature_size = 4096
hidden_size = 256
max_frames = 25
nbits = 64
transformer_type = 'small'

is_use_gpu = True
is_use_parallel = False
gpu_id = 0
is_use_softmax = True

#Visodom
env = "train"
is_vis = True

# dataset
dataset = 'fcv'
workers = 6   # origin 1
batch_size = 512   # origin 512
mask_prob = 0.75

# train
seed = 1
num_epochs = 805
a = 10
temperature = 0.5
tau_plus = 0.1
train_num_sample = 45585

# test
test_batch_size = 256
test_num_sample = 45600

# optimizer
optimizer_name = 'Adam'
schedule = 'StepLR'
lr = 3e-4
min_lr = 1e-5
lr_decay_rate = 100
lr_decay_gamma = 0.9
weight_decay = 0.0

e_every = 1
d_every = 1

# NCE parameters
K= 4096
T= 0.5   #me,20241216
momentum= 0.85
warmup_epoch = 3

# path
# data_root = '/home/yantk/Data/fcv/'
data_root = '../../../Data/video_datasets/fcv/'
home_root = './checkpoint/'

train_feat_path = [data_root + 'fcv_train_feats.h5']
test_feat_path = [data_root + 'fcv_test_feats.h5']
label_path = [data_root + 'fcv_test_labels.mat']

save_dir = home_root  + model_name + '_' + dataset
file_path = save_dir + '_' + str(nbits) + 'bit'